package com.utn.disenio.entrega3;

import java.util.ArrayList;
import java.util.List;

public class FiltroDeClases  implements TipoDeFiltro  {
	private List<String> clases;
	
	public FiltroDeClases(List<String> clases){
		this.clases = clases;
	}
	
	public ArrayList<Viaje> filtrar(ArrayList<Viaje> viajesAFiltrar) {
		ArrayList<Viaje> listaFiltrada = new ArrayList<Viaje>();
		for(Viaje unViaje: viajesAFiltrar)
		{
			if(this.clases.contains(unViaje.getAsientoUno().getClase()) &&
					  (unViaje.getAsientoDos() != null && this.clases.contains(unViaje.getAsientoDos().getClase())) &&
					  (unViaje.getAsientoTres() != null && this.clases.contains(unViaje.getAsientoTres().getClase()))
					)
						listaFiltrada.add(unViaje);
		}
		return listaFiltrada;
	}

}
